Checking For Prime Numbers In One Line Of Python